import unittest
import directed_acyclic_graph as dag


class MyTestCase(unittest.TestCase):
    def test_something(self):
        dagraph = dag.DAG()
        (dagraph.add_edge("A", "B", 5).add_edge("A", "C", 2).add_edge("C", "B", 8).add_edge("B", "D", 4).add_edge("B",
                                                                                                                  "E",
                                                                                                                  2)
         .add_edge("C", "E", 7).add_edge("D", "E", 6).add_edge("D", "F", 3).add_edge("E", "F", 1))
        print(dagraph.dijkstra("A", "F"))
        # self.assertEqual(True, False)  # add assertion here


if __name__ == '__main__':
    unittest.main()
